package com.finalBlog.common;

import javax.sql.DataSource;

import com.jfinal.kit.PathKit;
import com.jfinal.kit.Prop;
import com.jfinal.kit.PropKit;
import com.jfinal.plugin.activerecord.dialect.MysqlDialect;
import com.jfinal.plugin.activerecord.generator.Generator;
import com.jfinal.plugin.druid.DruidPlugin;

/**
 * finalBlog
 *	
 * 类/接口信息描述:jfianl model bean 自动生成器,添加新表或修改表信息后重新运行即可
 *
 * @author dengxc
 *
 */
public class GeneratorModel {
	public static void main(String[] args) {
		DataSource dataSource =getDataSource();
		// base model 所使用的包名
		String baseModelPkg = "com.finalBlog.model.base";
		// base model 文件保存路径
		String baseModelDir = PathKit.getWebRootPath() + "/src/main/java/com/finalBlog/model/base";
		// model 所使用的包名
		String modelPkg = "com.finalBlog.model";
		// model 文件保存路径
		String modelDir = baseModelDir + "/..";
		Generator gernerator = new Generator(dataSource, baseModelPkg, baseModelDir,
				modelPkg, modelDir);
		gernerator.setDialect(new MysqlDialect());
		// 添加不需要生成的表名
		//gernerator.addExcludedTable("adv");
		// 设置是否在 Model 中生成 dao 对象
		gernerator.setGenerateDaoInModel(true);
		// 设置是否生成字典文件
		gernerator.setGenerateDataDictionary(true);
		// 设置需要被移除的表名前缀用于生成modelName。例如表名 "osc_user"，移除前缀 "osc_"后生成的model名为 "User"而非 OscUser
		gernerator.setRemovedTableNamePrefixes("tb_");
		gernerator.generate();
	}
	public static DataSource getDataSource() {
		Prop prop = PropKit.use("baseConfig.properties");
		DruidPlugin druidPlugin = new DruidPlugin(
				prop.get("db.jdbcUrl"), 
				prop.get("db.userName"),
				prop.get("db.passWord"),
				prop.get("db.driver"));
		druidPlugin.start();
		return druidPlugin.getDataSource();
	}
}
